# %load AI/code/models/criterions.py
# 损失函数
# 多分类（cross entropy）
criterion = nn.CrossEntropyLoss()
# 多标签分类（binary cross entropy）
# 语义分割（binary cross entropy
# 如果是多类语义分割，可选cross entropy）

# criterion = nn.NLLLoss()
